../../../../models/demos/vanilla_unet/tests/test_unet_model.py